Baselines and Score Centering in Policy Gradient Estimation
策略梯度估计中的基线项与分数中心化
🧭 RLHF/策略梯度估计·重要性采样
University of Illinois Chicago · Perplexity AI
用 reward 与 importance-weighted score 的协方差替代正则化 IS 策略梯度,厘清 group-mean baseline 与 score centering 的不同作用
📌 工作参考价值(4 点)
- 在电商/Agent 的 LLM RLHF 后训练中,若用 sequence-level PPO/GRPO 且 ratio 被 clip,直接用 batch 内 mean reward 作为 group-mean baseline 即可,不必再对 importance-weighted score 做额外 centering,降低实现复杂度。
- 若采用 token-level advantage(细粒度 reward/PPO),当 importance ratio 被 truncation/masking/clipping 时,必须对 importance-weighted score 做 score centering(减去总体均值,可用 moving average 估计),否则策略梯度会带偏置。
- 建议把梯度写成 Cov(reward, weighted_score) 而非单纯加权 reward;该形式天然对 constant reward shift 稳健,适合离线异步训练中 stale policy 的纠偏。
- 序列级 IS 下 score centering 无额外收益,token 级 IS 下则是必要修正;实现前先确认 reward/advantage 的粒度再决定是否引入 score centering。